//
//  UINavigationController+FinishBlock.m
//  

#import "UINavigationController+FinishBlock.h"

@implementation UINavigationController (FinishBlock)

- (VoidBlock)finishBlock {
    return objc_getAssociatedObject(self, @selector(finishBlock));
}

- (void)setFinishBlock:(VoidBlock)aFinishBlock{
    objc_setAssociatedObject(self, @selector(finishBlock), aFinishBlock, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}



@end

@implementation UIViewController (FinishBlock)

- (VoidBlock)finishBlock {
    return objc_getAssociatedObject(self, @selector(finishBlock));
}

- (void)setFinishBlock:(VoidBlock)aFinishBlock{
    objc_setAssociatedObject(self, @selector(finishBlock), aFinishBlock, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}



@end
